# 修正导入语句，从utils.mongo导入db
from app.utils.mongo import db
from datetime import datetime

class Building:
    # 绑定MongoDB集合（使用正确的db实例）
    collection = db.buildings  # 这里从db获取buildings集合

    @classmethod
    def get_by_id(cls, building_id):
        """通过ID获取建筑详情"""
        return cls.collection.find_one({"_id": building_id})

    @classmethod
    def get_recommended(cls, limit=6):
        """获取推荐建筑"""
        return cls.collection.find({"is_recommended": True}).limit(limit)